import pandas
import os

class Pandas_test():
    def __init__(self,filename:str):

        #当前文件的路径:os.path.dirname(__file__) ; 当前的上一级目录:os.path.dirname(os.path.dirname(__file__))
        self.path=os.path.dirname(os.path.dirname(__file__))

        #读取文件
        if filename.endswith('.xls') or filename.endswith('.xlsx'):
            self.table=pandas.read_excel(self.path + '/Data/' + filename,keep_default_na=False) #dtype=str,keep_default_na=False
        elif filename.endswith('.csv'):
            self.table=pandas.read_csv(self.path + '/Data/' + filename)
        else:
            print('请输入正确的文件名称,结尾以.xls,.xlsx,.scv')
            self.table=None

     #列表嵌套列表
    def file_list(self):
        if self.table is not None:
            return self.table.values.tolist()
        else:
            print('文件读取失败')
    #列表嵌套字典
    def file_dict(self):
        if self.table is not None:
            return [self.table.loc[i].to_dict() for i in self.table.index.values]
        else:
            print('文件读取失败')




if __name__ == '__main__':

    test=Pandas_test("添加发布会.csv")
    # test.read_file()
    # print(test.file_list())
    print(test.file_dict())
